// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <shlobj.h>
#include <windows.h>

#include "base/base_paths.h"
#include "base/environment.h"
#include "base/files/file_path.h"
#include "base/path_service.h"
#include "base/strings/utf_string_conversions.h"
#include "base/win/current_module.h"
#include "base/win/scoped_co_mem.h"
#include "base/win/windows_version.h"

using base::FilePath;

namespace base {

bool PathProviderWin(int key, FilePath* result)
{
    // We need to go compute the value. It would be nice to support paths with
    // names longer than MAX_PATH, but the system functions don't seem to be
    // designed for it either, with the exception of GetTempPath (but other
    // things will surely break if the temp path is too long, so we don't bother
    // handling it.
    wchar_t system_buffer[MAX_PATH];
    system_buffer[0] = 0;

    FilePath cur;
    switch (key) {
    case base::FILE_EXE:
        if (GetModuleFileName(NULL, system_buffer, MAX_PATH) == 0)
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::FILE_MODULE: {
        // the resource containing module is assumed to be the one that
        // this code lives in, whether that's a dll or exe
        if (GetModuleFileName(CURRENT_MODULE(), system_buffer, MAX_PATH) == 0)
            return false;
        cur = FilePath(system_buffer);
        break;
    }
    case base::DIR_WINDOWS:
        GetWindowsDirectory(system_buffer, MAX_PATH);
        cur = FilePath(system_buffer);
        break;
    case base::DIR_SYSTEM:
        GetSystemDirectory(system_buffer, MAX_PATH);
        cur = FilePath(system_buffer);
        break;
    case base::DIR_PROGRAM_FILESX86:
        if (base::win::OSInfo::GetInstance()->architecture() != base::win::OSInfo::X86_ARCHITECTURE) {
            if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILESX86, NULL,
                    SHGFP_TYPE_CURRENT, system_buffer)))
                return false;
            cur = FilePath(system_buffer);
            break;
        }
        // Fall through to base::DIR_PROGRAM_FILES if we're on an X86 machine.
    case base::DIR_PROGRAM_FILES:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL,
                SHGFP_TYPE_CURRENT, system_buffer)))
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::DIR_PROGRAM_FILES6432:
#if !defined(_WIN64)
        if (base::win::OSInfo::GetInstance()->wow64_status() == base::win::OSInfo::WOW64_ENABLED) {
            std::unique_ptr<base::Environment> env(base::Environment::Create());
            std::string programfiles_w6432;
            // 32-bit process running in WOW64 sets ProgramW6432 environment
            // variable. See
            // https://msdn.microsoft.com/library/windows/desktop/aa384274.aspx.
            if (!env->GetVar("ProgramW6432", &programfiles_w6432))
                return false;
            // GetVar returns UTF8 - convert back to Wide.
            cur = FilePath(UTF8ToWide(programfiles_w6432));
            break;
        }
#endif
        if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAM_FILES, NULL,
                SHGFP_TYPE_CURRENT, system_buffer)))
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::DIR_IE_INTERNET_CACHE:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_INTERNET_CACHE, NULL,
                SHGFP_TYPE_CURRENT, system_buffer)))
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::DIR_COMMON_START_MENU:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_PROGRAMS, NULL,
                SHGFP_TYPE_CURRENT, system_buffer)))
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::DIR_START_MENU:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_PROGRAMS, NULL,
                SHGFP_TYPE_CURRENT, system_buffer)))
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::DIR_APP_DATA:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_APPDATA, NULL, SHGFP_TYPE_CURRENT,
                system_buffer)))
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::DIR_COMMON_APP_DATA:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_APPDATA, NULL,
                SHGFP_TYPE_CURRENT, system_buffer)))
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::DIR_LOCAL_APP_DATA:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_LOCAL_APPDATA, NULL,
                SHGFP_TYPE_CURRENT, system_buffer)))
            return false;
        cur = FilePath(system_buffer);
        break;
    case base::DIR_SOURCE_ROOT: {
        FilePath executableDir;
        // On Windows, unit tests execute two levels deep from the source root.
        // For example:  chrome/{Debug|Release}/ui_tests.exe
        PathService::Get(base::DIR_EXE, &executableDir);
        cur = executableDir.DirName().DirName();
        break;
    }
    case base::DIR_APP_SHORTCUTS: {
        if (win::GetVersion() < win::VERSION_WIN8)
            return false;

//         base::win::ScopedCoMem<wchar_t> path_buf;
//         if (FAILED(SHGetKnownFolderPath(FOLDERID_ApplicationShortcuts, 0, NULL, &path_buf)))
//             return false;
// 
//         cur = FilePath(string16(path_buf));
        DebugBreak();
        break;
    }
    case base::DIR_USER_DESKTOP:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_DESKTOPDIRECTORY, NULL,
                SHGFP_TYPE_CURRENT, system_buffer))) {
            return false;
        }
        cur = FilePath(system_buffer);
        break;
    case base::DIR_COMMON_DESKTOP:
        if (FAILED(SHGetFolderPath(NULL, CSIDL_COMMON_DESKTOPDIRECTORY, NULL,
                SHGFP_TYPE_CURRENT, system_buffer))) {
            return false;
        }
        cur = FilePath(system_buffer);
        break;
    case base::DIR_USER_QUICK_LAUNCH:
        if (!PathService::Get(base::DIR_APP_DATA, &cur))
            return false;
        // According to various sources, appending
        // "Microsoft\Internet Explorer\Quick Launch" to %appdata% is the only
        // reliable way to get the quick launch folder across all versions of
        // Windows.
        // http://stackoverflow.com/questions/76080/how-do-you-reliably-get-the-quick-
        // http://www.microsoft.com/technet/scriptcenter/resources/qanda/sept05/hey0901.mspx
        cur = cur.Append(FILE_PATH_LITERAL("Microsoft"))
                  .Append(FILE_PATH_LITERAL("Internet Explorer"))
                  .Append(FILE_PATH_LITERAL("Quick Launch"));
        break;
    case base::DIR_TASKBAR_PINS:
        if (!PathService::Get(base::DIR_USER_QUICK_LAUNCH, &cur))
            return false;
        cur = cur.Append(FILE_PATH_LITERAL("User Pinned"))
                  .Append(FILE_PATH_LITERAL("TaskBar"));
        break;
    case base::DIR_IMPLICIT_APP_SHORTCUTS:
        if (!PathService::Get(base::DIR_USER_QUICK_LAUNCH, &cur))
            return false;
        cur = cur.Append(FILE_PATH_LITERAL("User Pinned"))
                  .Append(FILE_PATH_LITERAL("ImplicitAppShortcuts"));
        break;
    case base::DIR_WINDOWS_FONTS:
        if (FAILED(SHGetFolderPath(
                NULL, CSIDL_FONTS, NULL, SHGFP_TYPE_CURRENT, system_buffer))) {
            return false;
        }
        cur = FilePath(system_buffer);
        break;
    default:
        return false;
    }

    *result = cur;
    return true;
}

} // namespace base
